home *** CD-ROM | disk | FTP | other *** search
/ Macwelt 4 / Macwelt DVD 4.cdr / Entwickler / Mac-OS X / Pantomime / Source / LocalMessage.m / LocalMessage.m
Encoding:
Text File  |  2002-08-20  |  4.1 KB  |  173 lines
  1. /*
  2. **  LocalMessage.m
  3. **
  4. **  Copyright (c) 2001, 2002
  5. **
  6. **  Author: Ludovic Marcotte <ludovic@Sophos.ca>
  7. **
  8. **  This library is free software; you can redistribute it and/or
  9. **  modify it under the terms of the GNU Lesser General Public
  10. **  License as published by the Free Software Foundation; either
  11. **  version 2.1 of the License, or (at your option) any later version.
  12. **  
  13. **  This library is distributed in the hope that it will be useful,
  14. **  but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. **  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16. **  Lesser General Public License for more details.
  17. **  
  18. **  You should have received a copy of the GNU Lesser General Public
  19. **  License along with this library; if not, write to the Free Software
  20. **  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  21. */
  22.  
  23. #import <Pantomime/LocalMessage.h>
  24.  
  25. #import <Pantomime/Constants.h>
  26. #import <Pantomime/LocalFolder.h>
  27. #import <Pantomime/NSDataExtensions.h>
  28.  
  29. #import <string.h>
  30.  
  31. @implementation LocalMessage 
  32.  
  33. //
  34. // NSCoding protocol
  35. //
  36. - (void) encodeWithCoder: (NSCoder *) theCoder
  37. {
  38.   //NSDebugLog(@"Encoding LocalMessage...");
  39.   [super encodeWithCoder: theCoder];
  40.   
  41.   [theCoder encodeObject: [NSNumber numberWithLong: [self filePosition]]];
  42.   [theCoder encodeObject: [NSNumber numberWithLong: [self bodyFilePosition]]];
  43. }
  44.  
  45. - (id) initWithCoder: (NSCoder *) theCoder
  46. { 
  47.   self = [super initWithCoder: theCoder];
  48.   
  49.   //NSDebugLog(@"Decoding LocalMessage...");
  50.  
  51.   [self setFilePosition: [[theCoder decodeObject] longValue]];
  52.   [self setBodyFilePosition: [[theCoder decodeObject] longValue]];
  53.   
  54.   return self;
  55. }
  56.  
  57. //
  58. // access / mutation methods
  59. //
  60. - (long) filePosition
  61. {
  62.   return filePosition;
  63. }
  64.  
  65. - (void) setFilePosition: (long) theFilePosition
  66. {
  67.   filePosition = theFilePosition;
  68. }
  69.  
  70. - (long) bodyFilePosition
  71. {
  72.   return bodyFilePosition;
  73. }
  74.  
  75. - (void) setBodyFilePosition: (long) theBodyFilePosition
  76. {
  77.   bodyFilePosition = theBodyFilePosition;
  78. }
  79.  
  80. - (NSData *) rawSource
  81. {
  82.   NSMutableData *aMutableData;
  83.   LocalFolder *aLocalFolder;
  84.   char aLine[1024];
  85.  
  86.   FILE *aStream;
  87.   long mark;
  88.  
  89.   aLocalFolder = (LocalFolder *)[self folder];
  90.   aStream = [aLocalFolder stream];
  91.  
  92.   mark = ftell(aStream);
  93.  
  94.   if (fseek(aStream, [self filePosition], SEEK_SET) < 0)
  95.     {
  96.       NSDebugLog( @"Seek operation failed!" );
  97.       return nil;
  98.     }
  99.  
  100.   // We initialize our mutable data and our buffer
  101.   aMutableData = [[NSMutableData alloc] init];
  102.   memset(aLine, 0, 1024);
  103.  
  104.   while( fgets(aLine, 1024, aStream) != NULL && 
  105.      ftell(aStream) < ([self filePosition] + [self size]) )
  106.     {
  107.       [aMutableData appendBytes: aLine  length: strlen(aLine) ];
  108.       memset(aLine, 0, 1024);
  109.     }
  110.  
  111.   fseek(aStream, mark, SEEK_SET);
  112.  
  113.   return AUTORELEASE(aMutableData);
  114. }
  115.  
  116. //
  117. // This method is called to initialize the message if it wasn't.
  118. // If we set it to NO and we HAD a content, we release the content;
  119. //
  120. - (void) setInitialized: (BOOL) aBOOL
  121. {
  122.   [super setInitialized: aBOOL];
  123.  
  124.   if ( aBOOL )
  125.     {
  126.       NSMutableData *aMutableData;
  127.       LocalFolder *aLocalFolder;
  128.       FILE *aStream;
  129.       char aLine[1024];
  130.       
  131.       int lengthOfBody;
  132.       long mark;
  133.       
  134.       
  135.       aLocalFolder = (LocalFolder *)[self folder];
  136.       aStream = [aLocalFolder stream];
  137.       
  138.       mark = ftell(aStream);
  139.       
  140.       if (fseek(aStream, [self bodyFilePosition], SEEK_SET) < 0)
  141.     {
  142.       NSDebugLog( @"Seek operation failed!" );
  143.       [super setInitialized: NO];
  144.       return;
  145.     }
  146.      
  147.       // We calculate the length of our body to make parsing faster
  148.       lengthOfBody = ([self filePosition] + [self size] - [self bodyFilePosition]);
  149.       aMutableData = [[NSMutableData alloc] initWithCapacity: lengthOfBody];
  150.       
  151.       memset(aLine, 0, 1024);
  152.       while( fgets(aLine, 1024, aStream) != NULL && 
  153.          ftell(aStream) < ([self filePosition] + [self size]) )
  154.     {
  155.       [aMutableData appendBytes: aLine  length: strlen(aLine) ];
  156.       memset(aLine, 0, 1024);
  157.     }
  158.       
  159.       
  160.       fseek(aStream, mark, SEEK_SET);
  161.       
  162.       [self setContentFromRawSource: aMutableData];
  163.       RELEASE(aMutableData);
  164.     }
  165.   else
  166.     {
  167.       TEST_RELEASE(content);
  168.       content = nil;
  169.     }
  170. }
  171.  
  172. @end
  173.